import click
from flask.cli import with_appcontext
from database import init_db, reset_database

def register_commands(app):
    """注册CLI命令"""
    
    @app.cli.command()
    @with_appcontext
    def init_db_command():
        """初始化数据库"""
        init_db(app)
        click.echo('数据库初始化完成！')
    
    @app.cli.command()
    @with_appcontext
    def reset_db_command():
        """重置数据库"""
        reset_database(app)
        click.echo('数据库重置完成！')
